# -*- coding: utf-8 -*-
# @Time    : 2021/7/7 14:46
# @Author  : LuoTianHang

# ####################config.py 说明##########################
# this script is the config script for the whole project

import argparse

def parser_config():
    parser = argparse.ArgumentParser(description="style transfer training")
    parser.add_argument("--project_name", type=str, default="style transfer training")
    parser.add_argument("--use_cuda", type=bool, default=True)
    parser.add_argument("--seed", type=int, default=123)

    parser.add_argument("--train_batch_size",type=int,default=2)
    parser.add_argument("--test_batch_size",type=int,default=2)

    parser.add_argument("--lr", type=float, default=0.001)
    parser.add_argument("--epochs", type=int, default=200)

    return parser.parse_args()